package com.yyds.oneservice.interceptor;

import com.yyds.common.CheckResult;
import com.yyds.common.R;
import com.yyds.common.constant.SystemConstant;
import com.yyds.common.utils.JwtUtils;
import com.yyds.oneservice.annotation.UnInterception;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.lang.StringUtils;
import org.springframework.http.MediaType;
import org.springframework.web.method.HandlerMethod;
import org.springframework.web.servlet.HandlerInterceptor;
import org.springframework.web.servlet.ModelAndView;

import javax.servlet.http.HttpServletRequest;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.io.PrintWriter;
import java.lang.reflect.Method;

/**
 * @date 2022/10/26
 * @desc   在这拦截器中  进行 token 校验
 */
@Slf4j    // 加了这个注解  可以直接使用  log 对象输出日志
public class MyInterceptor implements HandlerInterceptor {

    @Override
    public boolean preHandle(HttpServletRequest request, HttpServletResponse response, Object handler) throws Exception {
        /**
         * 使用场景： 如果某些接口需要登录后才能访问  可以在这里做登录校验
         *
         * */

        System.out.println(request.getRequestURI());  // 输出请求路径
        System.out.println(request.getPathInfo());
        if(handler instanceof HandlerMethod){
            // 如果接口方法处理器  就会走这里
            HandlerMethod handlerMethod = (HandlerMethod) handler;
            Method method = handlerMethod.getMethod();
            String methodName = method.getName();
            log.info("====拦截到了方法：{}，在该方法执行之前执行====", methodName);

            //得到方法上的注解 根据注解判断 这个方法需不需要拦截
            // 使用UnInterception 注解标注的方法 直接放行
            UnInterception annotation = method.getAnnotation(UnInterception.class);
            if(null != annotation){
                return true;
            }


            String token = request.getHeader("token");
            if(StringUtils.isEmpty(token)){
                log.info("token为空,请先登录");
                R r = R.error().code(SystemConstant.JWT_ERRCODE_NULL).message("token为空");
                // 向前端发送相应结果
                print(response,r);
                return false;
            }else{
                CheckResult checkResult = JwtUtils.validateJWT(token);

                if(checkResult.isSuccess()){
                    return true;
                }else{
                    switch (checkResult.getErrCode()){
                        case SystemConstant.JWT_ERRCODE_FAIL:{
                            System.out.println("token校验不通过");
                            R r = R.error().code(SystemConstant.JWT_ERRCODE_FAIL).message("token校验不通过");
                            // 向前端发送相应结果
                            print(response,r);
                            return false;
                        }

                        case SystemConstant.JWT_ERRCODE_EXPIRE:{
                            System.out.println("token校验不通过");
                            R r = R.error().code(SystemConstant.JWT_ERRCODE_EXPIRE).message("token已过期");
                            // 向前端发送相应结果
                            print(response,r);
                            return false;
                        }

                    }
                }
            }

        }
        // 如果是 资源处理器  比如 ResourceHttpRequestHandler  就不走上面的if

        return true;
    }

    @Override
    public void postHandle(HttpServletRequest request, HttpServletResponse response, Object handler, ModelAndView modelAndView) throws Exception {

        System.out.println("后置拦截，执行了controller 中的方法之后 响应视图之前 会执行这个方法");
    }

    @Override
    public void afterCompletion(HttpServletRequest request, HttpServletResponse response, Object handler, Exception ex) throws Exception {
        System.out.println("整个请求执行完毕 可以做一些收尾工作，如果请求过程中出现异常 这里也可以得到异常信息");
    }



    public void print(HttpServletResponse response,R r){

        response.setContentType(MediaType.APPLICATION_JSON_UTF8_VALUE);
        PrintWriter writer = null;
        try {
            writer = response.getWriter();
            writer.println(r);

            writer.flush();
            writer.close();
        } catch (IOException e) {
            e.printStackTrace();
        }


    }


}
